package com.wanghlme.bookstore.bean;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

public class OnLineBean {
    private Map<String, String> onLineUserMap = new HashMap<>();
    private Set<String> outLineUserSet = new HashSet<>();

    public Map<String, String> getOnLineUserMap() {
        return onLineUserMap;
    }

    public void setOnLineUserMap(Map<String, String> onLineUserMap) {
        this.onLineUserMap = onLineUserMap;
    }

    public Set<String> getOutLineUserSet() {
        return outLineUserSet;
    }

    public void setOutLineUserSet(Set<String> outLineUserSet) {
        this.outLineUserSet = outLineUserSet;
    }

    public int countUser() {
        return onLineUserMap.size();
    }

    public int countAllUser() {
        return onLineUserMap.size() + outLineUserSet.size();
    }

    public void addOutLineUserSet(String ip) {
        outLineUserSet.add(ip);
    }

    public void addOnLineUserSet(String ip, String username) {
        onLineUserMap.put(ip, username);
        outLineUserSet.remove(ip);
    }

    public void removeOnLineUserSet(String ip, boolean isOnline) {
        onLineUserMap.remove(ip);
        if (isOnline) {
            outLineUserSet.add(ip);
        } else {
            outLineUserSet.remove(ip);
        }

    }
}
